# -*- coding: utf-8 -*-
# !/usr/bin/env python
"""
-------------------------------------------------
   File Name：     config
   Description :   
   Author :       lth
   date：          2022/12/9
-------------------------------------------------
   Change Activity:
                   2022/12/9 9:14: create this script
-------------------------------------------------
"""
__author__ = 'lth'


class GetConfig:
    project_name = "NICE FLOW"
    use_cuda = True
    seed = 123
    resume = True
    pretrained_weight = "./weights/best.pth"
    lr = 1e-3
    epochs = 100

    train_batch_size = 1
    test_batch_size = 1
